package com.pb.infra.oth.util;

import java.io.*;

public class FileBytetransfer {

    //将文件转换成Byte数组
    public static byte[] getBytesByFile(InputStream inputStream) {

        try {

            ByteArrayOutputStream bos = new ByteArrayOutputStream(1000);
            byte[] b = new byte[1000];
            int n;
            while ((n = inputStream.read(b)) != -1) {
                bos.write(b, 0, n);
            }
            inputStream.close();
            byte[] data = bos.toByteArray();
            bos.close();
            return data;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

////    //将文件转换成Byte数组
//    public static byte[] getBytesByFile(OutputStream outputStream) {
//
//
//        try {
//            DrainableOutputStream drainableOutputStream=new DrainableOutputStream(outputStream);
//            ByteArrayOutputStream bos = new ByteArrayOutputStream(1000);
//            bos.writeTo(drainableOutputStream);
//
////            byte[] data = drainableOutputStream.toByteArray();
//            bos.close();
//            return data;
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return null;
//    }


    //将Byte数组转换成文件
    public static void getFileByBytes(byte[] bytes, String filePath, String fileName) {
        BufferedOutputStream bos = null;
        FileOutputStream fos = null;
        File file = null;
        try {
            File dir = new File(filePath);
            if (!dir.exists() && dir.isDirectory()) {// 判断文件目录是否存在
                dir.mkdirs();
            }
            file = new File(filePath + "\\" + fileName);
            fos = new FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (fos != null) {
                try {
                    fos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}
